package leetcode_滑动窗口;

import java.util.ArrayList;
import java.util.List;

/**
 * 找到字符串中所有字母异位词
 * 算法：滑动窗口 + 双指针
 *
 * @author yzh
 * @version 1.0
 * @date 2022/1/20 10:23
 */
public class _438 {
    public static void main(String[] args) {
        System.out.println(new _438().findAnagrams_upgrade("aaab", "aa"));
    }

    public List<Integer> findAnagrams(String s, String p) {
        int m = s.length(), n = p.length();
        int[] cnt = new int[26];
        for (int i = 0; i < n; i++) cnt[p.charAt(i) - 'a']++;
        List<Integer> ans = new ArrayList<>();
        int[] tmp = new int[26];
        for (int l = 0, r = 0; r < m; r++) {
            tmp[s.charAt(r) - 'a']++;
            if (r - l + 1 > n) tmp[s.charAt(l++) - 'a']--;
            if (check(cnt, tmp)) ans.add(l);
        }
        return ans;
    }

    boolean check(int[] a, int[] b) {
        for (int i = 0; i < 26; i++) if (a[i] != b[i]) return false;
        return true;
    }

    // check 方法主要是为了比较滑动窗口中的字符数量与 p 字符串中字符的数量是否相等, 这一步可以优化
    public List<Integer> findAnagrams_upgrade(String s, String p) {
        int m = s.length(), n = p.length();
        int[] cnt = new int[26];
        for (int i = 0; i < n; i++) cnt[p.charAt(i) - 'a']++;
        int a = 0;
        for (int i = 0; i < 26; i++) if (cnt[i] != 0) ++a;
        List<Integer> ans = new ArrayList<>();
        int b = 0;
        // 刚开始时, cnt 中只有 p 字符串中的字符的值大于 0
        // 没有在 p 字符串中出现的字符的 cnt 值不会大于 0
        for (int l = 0, r = 0; r < m; r++) {
            if (--cnt[s.charAt(r) - 'a'] == 0) ++b;
            if (r - l + 1 > n && ++cnt[s.charAt(l++) - 'a'] == 1) --b;
            if (a == b) ans.add(l);
        }
        return ans;
    }

}
